import os

from trainv7 import *

if __name__ == "__main__":
    opt = parse_opt()
    opt.weights = './weights/yolov7-tiny.pt'
    opt.data = './cjc/data/sex.yaml'
    # opt.img_size = [160,160]
    opt.resume = False
    opt.name = 'sex'
    # opt.epochs = 200
    # opt.batch_size = 3
    main(opt)
